import unittest

from count import count


class TestSub(unittest.TestCase):
    def test_sub1(self):
        c=count(2,3)
        r=c.sub()
        self.assertEqual(r, -1)
    def test_sub2(self):
        c=count(4,3)
        r=c.sub()
        self.assertEqual(r, 1)

    def test_sub3(self):
        c=count(6,3)
        r=c.sub()
        self.assertEqual(r, 3)


if __name__ == '__main__':
    unittest.main()
